from collections import OrderedDict

from rest_framework.pagination import PageNumberPagination


# 自定义分页器
from rest_framework.response import Response


class UserDefinePagination(PageNumberPagination):
    page_size = 10
    page_size_query_param = "pagesize"
    max_page_size = 20

    # 自定义分页返回内容格式
    def get_paginated_response(self, data):
        return Response(OrderedDict([
            ("count", self.page.paginator.count),
            ("lists", data),
            ("page", self.page.number),
            ("pages", self.page.paginator.num_pages),
            ("pagesize", self.page_size),
        ]))



def jwt_response_payload_handler(token, user=None, request=None):
    """
    自定义jwt认证成功返回数据
    """
    return {
        'token': token,
        'id': user.id,
        'username': user.username
    }